package com.atguigu.gmall.common.config;

import com.atguigu.gmall.common.constant.MqConst;
import com.rabbitmq.client.ConfirmCallback;
import lombok.SneakyThrows;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * @Author: zm
 * @Date: 2021/8/25 21:32
 */

/**
 * @Description 消息发送确认
 * <p>
 * ConfirmCallback  只确认消息是否正确到达 Exchange 中
 * ReturnCallback   消息没有正确到达队列时触发回调，如果正确到达队列不执行
 * <p>
 * 1. 如果消息没有到exchange,则confirm回调,ack=false
 * 2. 如果消息到达exchange,则confirm回调,ack=true
 * 3. exchange到queue成功,则不回调return
 * 4. exchange到queue失败,则回调return
 */
@Component
@Configuration
public class MQProducerAckConfig implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @PostConstruct // 修饰一个非静态的void（）方法,在服务器加载Servlet的时候运行，并且只会被服务器执行一次在构造函数之后执行，init（）方法之前执行。
    public void init() {
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * 确认消息是否发送到交换机
     *
     * @param correlationData 可以承载消息
     * @param ack             是否发送到交换机
     * @param cause           原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        if (ack) {
            System.out.println("消息发送成功================================");
        } else {
            System.out.println("消息发送不成======================");
        }
    }

    /**
     * 表示如果消息没有正确的到队列中，才会走这个方法！如果正确被消费了。这个方法是不会走的！
     *
     * @param message    消息主体
     * @param replyCode  状态码
     * @param replyText  失败详细
     * @param exchange   交换机
     * @param routingKey 路由key
     */
    @SneakyThrows
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {

        if (MqConst.ROUTING_ORDER_CANCEL.equals(routingKey)) {
            System.out.println("延时信息======================");
        } else {
            // 反序列化对象输出
            System.out.println("消息主体: " + new String(message.getBody(), "utf-8"));
            System.out.println("应答码: " + replyCode);
            System.out.println("描述：" + replyText);
            System.out.println("消息使用的交换器 exchange : " + exchange);
            System.out.println("消息使用的路由键 routing : " + routingKey);
        }


    }
}
